package com.skiving.cloud.admin.controller;

import cn.dev33.satoken.annotation.SaCheckPermission;
import cn.hutool.crypto.digest.BCrypt;
import com.mybatisflex.core.paginate.Page;
import com.skiving.cloud.admin.entity.*;
import com.skiving.cloud.admin.service.*;
import com.skiving.cloud.auth.entity.AuthUser;
import com.skiving.cloud.auth.entity.UserQueryParameter;
import com.skiving.cloud.auth.util.UserUtil;
import com.skiving.cloud.common.utils.OtherUtil;
import com.skiving.cloud.common.utils.PageParameter;
import com.skiving.cloud.common.utils.ReturnT;
import com.skiving.cloud.common.utils.UuidUtil;
import com.skiving.cloud.config.PasswordExpired;
import com.skiving.cloud.constant.AdminConstant;
import com.skiving.cloud.logutils.OperateLogLocal;
import lombok.extern.slf4j.Slf4j;
import org.noear.solon.annotation.*;
import org.noear.solon.core.util.MimeType;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;

import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 系统用户 控制层。
 *
 * @author skivingcloud
 * @since 2025-03-08
 */
@Slf4j
@Controller
@Api("系统用户接口")
@Mapping("/sysUser")
@Produces(MimeType.APPLICATION_JSON_UTF8_VALUE)
@Remoting
public class SysUserController {

    @Inject
    private SysUserService sysUserService;
    @Inject
    private SysRoleService sysRoleService;
    @Inject
    private SysStationService sysStationService;
    @Inject
    private SysUserRoleService sysUserRoleService;
    @Inject
    private SysUserStationService sysUserStationService;
    @Inject
    private SysMenuService sysMenuService;
    @Inject
    private SysUserMenuService sysUserMenuService;
    @Inject
    private PasswordExpired passwordExpired;
    /**
     * 添加系统用户。
     *
     * @param sysUser 系统用户
     * @return {@code true} 添加成功，{@code false} 添加失败
     */
    @Post
    @Mapping("/save")
    @ApiOperation("保存系统用户")
    @OperateLogLocal(module="系统管理-用户管理", description = "新增用户管理")
    @SaCheckPermission("sys:user:add")
    public ReturnT<Boolean> save(@Body @ApiParam("系统用户") SysUser sysUser) {
        sysUser.setId(UuidUtil.get32UUID());
        sysUser.setCreateUserId(UserUtil.getCurrentUser().getId());
        sysUser.setCreateTime(LocalDateTime.now());
        sysUser.setPassword(BCrypt.hashpw(passwordExpired.getDefaultValue()));
        return ReturnT.ok(sysUserService.save(sysUser));
    }

    /**
     * 根据主键删除系统用户。
     *
     * @param id 主键
     * @return {@code true} 删除成功，{@code false} 删除失败
     */
    @Post
    @Mapping("/remove/{id}")
    @ApiOperation("根据主键系统用户")
    @OperateLogLocal(module="系统管理-用户管理", description = "删除用户管理")
    @SaCheckPermission("sys:user:delete")
    @Produces(MimeType.APPLICATION_JSON_UTF8_VALUE)
    public ReturnT<Boolean> remove(@Path("id") @ApiParam("系统用户主键") String id) {
        if(AdminConstant.SUPER_ADMIN.equals(id)){
            return ReturnT.error("内置用户禁止删除！");
        }
        //逻辑删除
        SysUser sysUser = new SysUser();
        sysUser.setId(id);
        sysUser.setStatus(AdminConstant.USER_STATUS_DELETED);
        return ReturnT.ok(sysUserService.updateById(sysUser, true));
    }

    /**
     * 根据主键更新系统用户。
     *
     * @param sysUser 系统用户
     * @return {@code true} 更新成功，{@code false} 更新失败
     */
    @Post
    @Mapping("/update")
    @ApiOperation("根据主键更新系统用户")
    @OperateLogLocal(module="系统管理-用户管理", description = "修改用户管理")
    @SaCheckPermission("sys:user:edit")
    public ReturnT<Boolean> update(@Body @ApiParam("系统用户主键") SysUser sysUser) {
        return ReturnT.ok(sysUserService.updateById(sysUser, true));
    }

    /**
     * 查询所有系统用户。
     *
     * @return 所有数据
     */
    @Get
    @Mapping("/list")
    @ApiOperation("查询所有系统用户")
    public ReturnT<List<SysUser>> list() {
        return ReturnT.ok(sysUserService.list());
    }

    /**
     * 根据系统用户主键获取详细信息。
     *
     * @param id 系统用户主键
     * @return 系统用户详情
     */
    @Get
    @Mapping("/getInfo/{id}")
    @ApiOperation("根据主键获取系统用户")
    public ReturnT<SysUser> getInfo(@Path("id") @ApiParam("系统用户主键") String id) {
        return ReturnT.ok(sysUserService.getById(id));
    }

    /**
     * 分页查询系统用户。
     *
     * @param parameter 分页对象
     * @return 分页对象
     */
    @Post
    @Mapping("/page")
    @ApiOperation("分页查询系统用户")
    public ReturnT<Page<SysUser>> page(@Body @ApiParam("分页信息") PageParameter<SysUser> parameter) {
        return ReturnT.ok(sysUserService.queryByPage(parameter));
    }

    /**
     * 根据主键重置密码。
     *
     * @param id 主键
     * @return {@code 0} 重置密码成功，{@code 500} 重置密码失败
     */
    @Post
    @Mapping("/resetPassword/{id}")
    @ApiOperation("根据主键重置密码")
    @SaCheckPermission("sys:user:resetPassword")
    @OperateLogLocal(module="系统管理-用户管理", description = "管理员重置用户密码")
    public ReturnT<String> resetPassword(@Path("id") @ApiParam("主键")String id) {
        try{
            SysUser sysUser = sysUserService.getById(id);
            //生成一个带数字、字母的10位随机密码
            String newPassword = OtherUtil.generatePassword(10);
            sysUser.setPassword(BCrypt.hashpw(newPassword));
            sysUserService.updateById(sysUser, true);
            return ReturnT.ok(newPassword);
        }catch(Exception e){
            log.error("重置密码失败", e);
            return ReturnT.error("重置密码失败");
        }
    }

    /**
     * 根据用户id获取关联角色
     * @param id 用户id
     * @return 结果
     */
    @Get
    @Mapping("/getUserRolesByUserId/{id}")
    @ApiOperation("根据用户id获取关联角色")
    public ReturnT<Map<String, Object>> getUserRolesByUserId(@Path("id") @ApiParam("主键")String id) {
        try{
            SysUser sysUser = sysUserService.getById(id);
            List<SysRole> roles = sysRoleService.listByTenantId(sysUser.getTenantId());
            List<SysUserRole> userRoles = sysUserRoleService.listByUserId(id);
            Map<String, Object> map = new HashMap<>();
            map.put("roles", roles);
            map.put("userRoles", userRoles);
            return ReturnT.ok(map);
        }catch(Exception e){
            log.error("根据用户id获取关联角色失败", e);
            return ReturnT.error("根据用户id获取关联角色失败");
        }
    }

    /**
     * 设置用户角色
     * @param sysUser 用户
     * @return 修改结果
     */
    @ApiOperation("设置用户角色")
    @Post
    @Mapping("/setUserRolesByUserId")
    @SaCheckPermission("sys:user:role")
    @OperateLogLocal(module="系统管理-用户管理", description = "设置用户角色")
    public ReturnT<String> setUserRolesByUserId(@Body SysUser sysUser){
        try{
            sysUserService.setUserRolesByUserId(sysUser);
            return ReturnT.ok("设置用户角色成功");
        }catch(Exception e){
            log.error("设置用户角色失败", e);
            return ReturnT.error("设置用户角色失败");
        }
    }

    /**
     * 根据用户id获取关联岗位
     * @param id 用户id
     * @return 结果
     */
    @Get
    @Mapping("/getUserStationTree/{id}")
    @ApiOperation("根据用户id获取关联岗位")
    public ReturnT<Map<String, Object>> getUserStationTree(@Path("id") @ApiParam("主键")String id) {
        try{
            SysUser sysUser = sysUserService.getById(id);
            List<SysOrg> orgs = sysStationService.treeByTenantId(sysUser.getTenantId());
            List<SysUserStation> userStations = sysUserStationService.listByUserId(id);
            Map<String, Object> map = new HashMap<>();
            map.put("stations", orgs);
            map.put("userStations", userStations.stream().map(SysUserStation::getStationId).toList());
            return ReturnT.ok(map);
        }catch(Exception e){
            log.error("根据用户id获取关联岗位失败", e);
            return ReturnT.error("根据用户id获取关联岗位失败");
        }
    }

    /**
     * 设置用户岗位
     * @param sysUser 用户
     * @return 修改结果
     */
    @ApiOperation("设置用户岗位")
    @Post
    @Mapping("/setUserStationByUserId")
    @SaCheckPermission("sys:user:station")
    @OperateLogLocal(module="系统管理-用户管理", description = "设置用户岗位")
    public ReturnT<String> setUserStationByUserId(@Body SysUser sysUser){
        try{
            sysUserService.setUserStationsByUserId(sysUser);
            return ReturnT.ok("设置岗位权限成功");
        }catch(Exception e){
            log.error("设置岗位权限失败", e);
            return ReturnT.error("设置岗位权限失败");
        }
    }

    /**
     * 根据租户和用户id加载资源树权限
     *
     * @param tenantId 租户ID
     * @param id 角色ID
     * @return 资源权限
     */
    @ApiOperation("根据租户和用户id加载资源树权限")
    @Get
    @Mapping("/getTenantUserResTree/{tenantId}/{id}")
    public ReturnT<Map<String, Object>> getTenantUserResTree(@Path("tenantId") String tenantId, @Path("id") String id){
        AuthUser user = UserUtil.getCurrentUser();
        if (!user.getTenantId().equals(AdminConstant.SUPER_ADMIN_TENANT_ID)) {
            tenantId = user.getTenantId();
        }
        if (tenantId == null) {
            return ReturnT.error("租户不明确没数据");
        }
        List<SysMenu> list = sysMenuService.getTenantMenuList(tenantId);
        List<SysUserMenu> userResource = sysUserMenuService.listByUserId(id);
        Map<String, Object> res = new HashMap<>();
        res.put("tree", list);
        res.put("userResource", userResource.stream().map(SysUserMenu::getMenuId).toList());
        return ReturnT.ok(res);
    }

    /**
     * 设置用户资源
     * @param sysUser 用户
     * @return 修改结果
     */
    @ApiOperation("设置用户资源")
    @Post
    @Mapping("/setUserMenuByUserId")
    @SaCheckPermission("sys:user:menu")
    @OperateLogLocal(module="系统管理-用户管理", description = "设置用户资源权限")
    public ReturnT<String> setUserMenuByUserId(@Body SysUser sysUser){
        try{
            sysUserService.setUserMenuByUserId(sysUser);
            return ReturnT.ok("设置用户资源权限成功");
        }catch(Exception e){
            log.error("设置用户资源权限失败", e);
            return ReturnT.error("设置用户资源权限失败");
        }
    }

    /**
     * 根据岗位ID和用户ID查询用户姓名
     * @param parameter 参数
     * @return 逗号隔开的用户姓名
     */
    @Post
    @Mapping("/getUserNameByStationIdsAndUserIds")
    @ApiOperation("根据岗位ID和用户ID查询用户姓名")
    public ReturnT<String> getUserNameByStationIdsAndUserIds(@Body UserQueryParameter parameter){
        return ReturnT.ok(sysUserService.getUserNameByStationIdsAndUserIds(parameter));
    }

    /**
     * 获取本租户下用户信息{key，value}
     * @return 用户信息
     */
    @ApiOperation("查询本租户下用户信息")
    @Get
    @Mapping("/getAllUserBySelfTenant")
    public ReturnT<List<Map<String, Object>>> getAllUserBySelfTenant(){
        return ReturnT.ok(sysUserService.getAllUserBySelfTenant(UserUtil.getCurrentUser()));
    }

    /**
     * 获取本租户下用户ID
     * @return 用户信息
     */
    @ApiOperation("查询本租户下用户ID")
    @Get
    @Mapping("/getAllUserIdsBySelfTenant")
    public ReturnT<List<String>> getAllUserIdsBySelfTenant(){
        return ReturnT.ok(sysUserService.getAllUserIdsBySelfTenant(UserUtil.getCurrentUser()));
    }

    /**
     * 根据租户获取关联岗位
     * @param tenantId 用户租户
     * @return 结果
     */
    @Get
    @Mapping("/getUserStationTreeByTenant/{tenantId}")
    @ApiOperation("根据租户id获取关联岗位")
    public ReturnT<List<SysOrg>> getUserStationTreeByTenant(@Path("tenantId") @ApiParam("主键")String tenantId) {
        try{
            List<SysOrg> orgList = sysStationService.treeByTenantId(tenantId);
            return ReturnT.ok(orgList);
        }catch(Exception e){
            log.error("根据租户id获取关联岗位失败", e);
            return ReturnT.error("根据租户id获取关联岗位失败");
        }
    }

    /**
     * 根据用户ID获取岗位集合
     * @param userId 用户ID
     * @return 结果
     */
    @Get
    @Mapping("/getStationListByUserId/{userId}")
    @ApiOperation("根据用户ID获取岗位集合")
    public ReturnT<List<SysStation>> getStationListByUserId(@Path("userId") @ApiParam("用户ID")String userId) {
        try{
            List<SysStation> stationList = sysStationService.getStationListByUserId(userId);
            return ReturnT.ok(stationList);
        }catch(Exception e){
            log.error("根据用户ID获取岗位集合", e);
            return ReturnT.error("根据用户ID获取岗位集合");
        }
    }

    /**
     * 根据用户ID获取岗位集合
     * @param userId 用户ID
     * @return 结果
     */
    @Get
    @Mapping("/getStationIdListByUserId/{userId}")
    @ApiOperation("根据用户ID获取岗位集合")
    public List<String> getStationIdListByUserId(@Path("userId") @ApiParam("用户ID")String userId) {
        try{
            List<SysStation> stationList = sysStationService.getStationListByUserId(userId);
            return stationList.stream().map(SysStation::getId).toList();
        }catch(Exception e){
            log.error("根据用户ID获取岗位集合失败", e);
            return new ArrayList<>();
        }
    }

    /**
     * 根据用户ID获取岗位集合
     * @param userId 用户ID
     * @return 结果
     */
    @Get
    @Mapping("/getStationIdListByUserIdRpc")
    @ApiOperation("根据用户ID获取岗位集合")
    public List<String> getStationIdListByUserIdRpc(@ApiParam("用户ID")String userId) {
        try{
            List<SysStation> stationList = sysStationService.getStationListByUserId(userId);
            return stationList.stream().map(SysStation::getId).toList();
        }catch(Exception e){
            log.error("根据用户ID获取岗位集合失败", e);
            return new ArrayList<>();
        }
    }

    /**
     * 根据用户名查询用户信息
     * @param parameter 参数
     * @return 用户信息
     */
    @Post
    @Mapping("/queryUserByName")
    @ApiOperation("根据用户名查询用户信息")
    public ReturnT<List<SysUser>> queryUserByName(@Body PageParameter<SysUser> parameter){
        if (parameter.getParam() == null || parameter.getParam().getStaffName().isBlank()) {
            return ReturnT.ok();
        }
        parameter.getParam().setTenantId(UserUtil.getCurrentUser().getTenantId());
        return ReturnT.ok(sysUserService.queryByPage(parameter).getRecords());
    }
}
